package networks;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.SocketException;

// 回显服务器
public class UdpEchoServer {
    private DatagramSocket socket = null;

    public UdpEchoServer(int port) throws SocketException {
        // 指定了一个固定的端口号，让服务器来使用
        socket = new DatagramSocket(port);
    }

    public void start() throws IOException {
        // 启动服务器
        System.out.println("服务器启动...");

        // 不断读取客户端请求
        while(true){
            // 循环一次，相当于处理一次请求
            // 处理请求的三个步骤

            // 1. 读取请求并解析

            // 这里receive方法也是存在输出型参数
            // 先构造一个空的实例，作为参数传给receive方法，该方法莫得返回值，在方法内对其进行初始化
            // DatagramPocket表示一个UDP数据报，byte数组保存UDP的载荷部分
            DatagramPacket requestPacket = new DatagramPacket(new byte[4096], 4096);
            socket.receive(requestPacket);

            // 把读取到的二进制数据转为字符串
            // 这里的getData方法获得requestPacket中的字节数组
            String request = new String(requestPacket.getData(), 0, requestPacket.getLength());

            // 2. 根据请求处理相应（服务器最主要的逻辑）
            //      这里是回显服务器，因此忽略了这一部分
            String response = process(request);

            // 3. 把相应返回给客户端

            // 根据response构造responsePacket
            // 这里的getSocketAddress返回的就是源ip和源端口
            DatagramPacket responsePacket = new DatagramPacket(response.getBytes(), response.getBytes().length,
                    requestPacket.getSocketAddress());
            // 由于UDP没有保存对方的信息，不能直接返回，需要指定目的ip和目的端口
            // 谁发的请求，就给谁发相应，也就是请求的源ip和源端口
            socket.send(responsePacket);

            // 4.打印一个日志
            System.out.printf("[%s:%d] req: %s; resp: %s\n", requestPacket.getAddress().toString(),
                    responsePacket.getPort(), request, response);
        }
    }

    // 单独封装成方法，如果要实现不是回响的服务器，直接修改这里即可
    protected String process(String request) {
        return request;
    }

    public static void main(String[] args) throws IOException {
        UdpEchoServer server = new UdpEchoServer(9090);
        server.start();
    }
}
